#ifndef IM2COL_H
#define IM2COL_H

void im2col_cpu(float *data_im,
    int channels, int height, int width,
    int ksize, int stride, int pad, float *data_col);

#ifdef GPU

void im2col_gpu(float *im,
    int channel, int height, int width,
    int ksize, int stride, int pad, float *data_col);

#endif // GPU
#endif // IM2COL_H